package middle;

import java.util.Deque;
import java.util.LinkedList;

/**
 * 给定一个由 0 和 1 组成的矩阵 mat，请输出一个大小相同的矩阵，其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。
 * 两个相邻元素间的距离为 1 。
 * 链接：https://leetcode-cn.com/problems/01-matrix
 * @author 胡宇轩
 * @Email: yuxuan.hu01@bianlifeng.com
 */
public class ZeroOneMatrix {
    /**
     * BFS
     * https://leetcode-cn.com/problems/01-matrix/solution/2chong-bfs-xiang-jie-dp-bi-xu-miao-dong-by-sweetie/
     */
    class Solution {
        public int[][] updateMatrix(int[][] mat) {
            Deque<int[]> queue = new LinkedList<>();
            for (int i = 0; i < mat.length; i++) {
                for (int j = 0; j < mat[0].length; j++) {
                    if (mat[i][j] == 0) {
                        queue.offer(new int[]{i, j});
                    } else {
                        mat[i][j] = -1;
                    }
                }
            }
            int[] dx = new int[]{-1, 1, 0, 0};
            int[] dy = new int[]{0, 0, -1, 1};

            while (!queue.isEmpty()) {
                int[] poll = queue.poll();
                int x = poll[0];
                int y = poll[1];
                for (int i = 0; i < 4; i++) {
                    int newX = x + dx[i];
                    int newY = y + dy[i];
                    if (newX >= 0 && newX < mat.length && newY >= 0 && newY < mat[0].length && mat[newX][newY] == -1) {
                        mat[newX][newY] = mat[x][y] + 1;
                        queue.offer(new int[]{newX, newY});
                    }
                }
            }
            return mat;
        }
    }
}
